# coding:utf-8
"""
Author: 11203
Date:  16:51
Name: request_decorators
"""
from flask import g


# 自定义装饰器，用于（某些请求）验证请求是否已经登录
# 如果已经登录则可以访问，否则不运行访问
# 即登录拦截
def login_required(func):
    def wrapper_func(*args, **kwargs):

        # 符合该条件，则表示已经登录
        if g.user_id is not None:
            return func(*args, **kwargs)
        else:
            return {'message': '非法的token'}

    return wrapper_func
